"""
__author__ = 'berlinyoung'
@File   :main_page.py
@Date   :2022/5/9 16:48
-*- coding: utf-8 -*-
"""
from selenium.webdriver.common.by import By

from weixin_web.step1.po.add_member_page import AddMemberPage
from weixin_web.step1.po.base import Base


class MainPage(Base):
    _base_url = 'https://work.weixin.qq.com/wework_admin/frame#index'
    def goto_add_member(self):
        '''
        跳转到添加成员页面
        :return:
        '''
        self.find_and_click(By.CSS_SELECTOR,".ww_indexImg_AddMember")
        return AddMemberPage(self.driver)

    def goto_contact(self):
        '''
        跳转到通讯里页面
        :return:
        '''
        from weixin_web.step1.po.contact_page import ContactPage
        return ContactPage(self.driver)